Best way to handle if-else with argparse